import pandas as pd
import mysql.connector
from sqlalchemy import create_engine

# 读取 XLS 文件
xls_file = 'C:/Users/xuzhi/Desktop/breakfast.xls'
df = pd.read_excel(xls_file, sheet_name='Sheet1')

# MySQL 数据库连接配置
db_config = {
    'user': 'root',
    'password': '123456',
    'host': 'localhost',
    'database': 'fastapi',
    'raise_on_warnings': True
}

# 使用 mysql-connector 进行数据库连接
conn = mysql.connector.connect(**db_config)

# 创建一个 MySQL 数据库引擎
engine = create_engine('mysql+mysqlconnector://{user}:{password}@{host}/{database}'.format(**db_config))


# 将数据写入数据库中的表
table_name = 'test'
df.to_sql(table_name, engine, index=False, if_exists='replace')

# 关闭数据库连接
conn.close()
